package com.threez.single.admin.singleadmin;

import com.baomidou.mybatisplus.annotation.FieldFill;
import com.baomidou.mybatisplus.generator.FastAutoGenerator;
import com.baomidou.mybatisplus.generator.config.OutputFile;
import com.baomidou.mybatisplus.generator.fill.Column;
import com.sun.istack.internal.NotNull;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.List;

/**
 * 代码生成器
 * @author zhaozz
 * @title MybatisGenerater
 * @date 2023/2/18 14:51
 */
public class MybatisGenerator {
    public static void main(String[] args) {

        FastAutoGenerator.create("jdbc:mysql://127.0.0.1:3306/id_sms?serverTimezone=GMT%2B8&useSSL=FALSE&useUnicode=true&characterEncoding=utf8&rewriteBatchedStatements=true", "root", "")
                // 全局配置
                .globalConfig((scanner, builder) -> builder.author(scanner.apply("请输入作者名称？"))
                        .outputDir(System.getProperty("user.dir") + "/src/main/java") // 指定输出目录
                        .disableOpenDir() //不会自动打开目录
                ) // 覆盖已生成文件
                // 包配置
                .packageConfig((scanner, builder) -> builder
//                        .moduleName("") // 设置父包模块名
                        .pathInfo(Collections.singletonMap(OutputFile.xml, System.getProperty("user.dir")+"/src/main/resources/mapper")) // 设置mapperXml生成路径 // 设置mapperXml生成路径
                        .parent("com.ids.sms")) // 设置父包名

                // 策略配置
                .strategyConfig((scanner, builder) -> builder.addInclude(getTables(scanner.apply("请输入表名，多个英文逗号分隔？所有输入 all")))
                        .controllerBuilder().enableRestStyle().enableHyphenStyle()
                        .enableFileOverride()
                        .entityBuilder().enableLombok().addTableFills(
                                new Column("create_time", FieldFill.INSERT)
                        ).build())
                /*
                    模板引擎配置，默认 Velocity 可选模板引擎 Beetl 或 Freemarker
                   .templateEngine(new BeetlTemplateEngine())
                   .templateEngine(new FreemarkerTemplateEngine())
                 */
                .execute();

    }

    // 处理 all 情况
    protected static @NotNull List<String> getTables(String tables) {
        return "all".equals(tables) ? new ArrayList<String>() : Arrays.asList(tables.split(","));
    }
}
